package com.chmingx.springcloud.controller;

import com.alibaba.csp.sentinel.annotation.SentinelResource;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.concurrent.TimeUnit;

@RestController
public class FlowLimitController {
    @GetMapping("/testA")
    public String testA() {
        try {
            TimeUnit.MILLISECONDS.sleep(800);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return "------ testA";
    }

    @GetMapping("/testB")
    public String testB() {
        return "------ testB";
    }

    // 热点限流，并配置违规后的处理方法
    @GetMapping("/testHotKey")
    @SentinelResource(value = "testHotKey", blockHandler = "dealTestHotKey")
    public String testHotKey(@RequestParam(value = "p1", required = false) String p1,
                             @RequestParam(value = "p2", required = false) String p2) {
        return "------ test Hot Key";
    }

    // 违规后的处理办法
    public String dealTestHotKey(String p1, String p2, BlockException exception) {
        return "------ test Hot Key, blocked"; //sentinel系统默认的提示：Blocked by Sentinel (flow limiting)
    }
}
